<?php
/*---------------------------------------------
 * @数据库备份还原控制器
 * @第一：本控制器依靠config中的配置运行：
 *		'DB_PATH_NAME'=> 'db',        //备份目录名称,主要是为了创建备份目录；
 *		'DB_PATH'     => './db/',     //数据库备份路径必须以 / 结尾；
 *		'DB_PART'     => '20971520',  //该值用于限制压缩后的分卷最大长度。单位：B；建议设置20M
 *		'DB_COMPRESS' => '1',         //压缩备份文件需要PHP环境支持gzopen,gzwrite函数        0:不压缩 1:启用压缩
 *		'DB_LEVEL'    => '9',         //压缩级别   1:普通   4:一般   9:最高
 * @第二：本控制器依赖ThinkPHP/Library/OT/Database.class.class.php
 * @第三：模版，Home/View/Database下面的2个文件 export.html备份数据库  import.html还原数据库
 * @第四：引用方法：<a href="{:U('Database.class/index',array('type'=>'export'))}">备份数据库</a>
 * 		          <a href="{:U('Database.class/index',array('type'=>'import'))}">还原数据库</a>
 * Author: 枫LT <957987132@qq.com>
 *---------------------------------------------
 */

namespace Admin\Controller;
use Think\Db;
use OT\Database;

/**
 * 数据库备份还原控制器
 */
class DatabaseController extends BaseController{

    /**
     * 数据库备份/还原列表
     * @param  String $type import-还原，export-备份
     */
    public function index($type = null){
        switch ($type) {
            /* 数据还原 */
            case 'import':
                //判断目录是否存在
                is_writeable($config['path']) || mkdir('./'.C("DB_PATH_NAME").'',0777,true);
                //列出备份文件列表
                $path = realpath(C('DB_PATH'));
                $flag = \FilesystemIterator::KEY_AS_FILENAME;
                $glob = new \FilesystemIterator($path,  $flag);

                $list = array();
                foreach ($glob as $name => $file) {
                    if(preg_match('/^\d{8,8}-\d{6,6}-\d+\.sql(?:\.gz)?$/', $name)){
                        $name = sscanf($name, '%4s%2s%2s-%2s%2s%2s-%d');

                        $date = "{$name[0]}-{$name[1]}-{$name[2]}";
                        $time = "{$name[3]}:{$name[4]}:{$name[5]}";
                        $part = $name[6];

                        if(isset($list["{$date} {$time}"])){
                            $info = $list["{$date} {$time}"];
                            $info['part'] = max($info['part'], $part);
                            $info['size'] = $info['size'] + $file->getSize();
                        } else {
                            $info['part'] = $part;
                            $info['size'] = $file->getSize();
                        }
                        $extension        = strtoupper(pathinfo($file->getFilename(), PATHINFO_EXTENSION));
                        $info['compress'] = ($extension === 'SQL') ? '-' : $extension;
                        $info['time']     = strtotime("{$date} {$time}");

                        $list["{$date} {$time}"] = $info;
                    }
                }
                $title = '数据还原';
                //记录操作到日志
                $action_name = "进行了一次数据还原";
                $time = date("Y-m-d H:i:s", time());
                parent::rizhi($action_name, $time);

                break;

            /* 数据备份 */
            case 'export':
                //记录操作到日志
                $action_name = "进行一次数据备份";
                $time = date("Y-m-d H:i:s", time());
                parent::rizhi($action_name, $time);

                $Db    = Db::getInstance();
                $list  = $Db->query('SHOW TABLE STATUS');
                $list  = array_map('array_change_key_case', $list);
                $title = '数据备份';
                break;

            default:
                $this->error('参数错误!');
        }
        //渲染模板
        $this->assign('meta_title', $title);
        $this->assign('list', $list);
        $this->display($type);
    }

    /**
     * 优化表
     * @param  String $tables 表名
     */
    public function optimize($tables = null){
        if($tables) {
            $Db   = Db::getInstance();
            if(is_array($tables)){
                $tables = implode('`,`', $tables);
                $list = $Db->query("OPTIMIZE TABLE `{$tables}`");

                if($list){
                    //记录操作到日志
                    $action_name = "完成了一次对".$tables."表的优化";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    $this->ajaxReturn(['info'=>'数据表优化完成!','status'=>1]);
                } else {
                    $this->ajaxReturn(['info'=>'数据表优化出错请重试!','status'=>0]);
                }
            } else {
                $list = $Db->query("OPTIMIZE TABLE `{$tables}`");
                if($list){
                    //记录操作到日志
                    $action_name = "完成了一次对".$tables."表的优化";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    //$this->success("数据表'{$tables}'优化完成!");
                    $this->ajaxReturn(['info'=>"数据表'{$tables}'优化完成!",'status'=>0]);
                } else {
                    //$this->error("数据表'{$tables}'优化出错请重试!");
                    $this->ajaxReturn(['info'=>"数据表'{$tables}'优化出错请重试!",'status'=>0]);
                }
            }
        } else {
            //$this->error("请指定要优化的表!");
            $this->ajaxReturn(['info'=>"请指定要优化的表!",'status'=>1]);
        }
    }

    /**
     * 修复表
     * @param  String $tables 表名
     */
    public function repair($tables = null){
        if($tables) {
            $Db   = Db::getInstance();
            if(is_array($tables)){
                $tables = implode('`,`', $tables);
                $list = $Db->query("REPAIR TABLE `{$tables}`");

                if($list){
                    //记录操作到日志
                    $action_name = "完成了一次对".$tables."表的修复";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    //$this->success("数据表修复完成!");
                    $this->ajaxReturn(['info'=>"数据表修复完成!",'status'=>1]);
                } else {
                    //$this->error("数据表修复出错请重试!");
                    $this->ajaxReturn(['info'=>"数据表修复出错请重试!",'status'=>0]);
                }
            } else {
                $list = $Db->query("REPAIR TABLE `{$tables}`");
                if($list){
                    //记录操作到日志
                    $action_name = "完成了一次对".$tables."表的修复";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    //$this->success("数据表'{$tables}'修复完成!");
                    $this->ajaxReturn(['info'=>"数据表'{$tables}'修复完成!",'status'=>0]);
                } else {
                    //$this->error("数据表'{$tables}'修复出错请重试!");
                    $this->ajaxReturn(['info'=>"数据表'{$tables}'修复出错请重试!",'status'=>0]);
                }
            }
        } else {
            //$this->error("请指定要修复的表!");
            $this->ajaxReturn(['info'=>"请指定要修复的表!",'status'=>0]);
        }
    }

    /**
     * 删除备份文件
     * @param  Integer $time 备份时间
     */
    public function del($time = 0){
        if($time){
            $name  = date('Ymd-His', $time) . '-*.sql*';
            $path  = realpath(C('DB_PATH')). DIRECTORY_SEPARATOR . $name;
            array_map("unlink", glob($path));
            $path1 =C('DB_PATH').'backup.lock';
            array_map('unlink',glob($path1));
            if(count(glob($path))){
                //$this->success('备份文件删除失败，请检查权限!');
                $this->ajaxReturn(['info'=>'备份文件删除失败，请检查权限!','status'=>0]);
            } else {

                //记录操作到日志
                $action_name = "完成了一次对备份时间为".$time."的备份文件的删除";
                $time = date("Y-m-d H:i:s", time());
                parent::rizhi($action_name, $time);

                $this->ajaxReturn(['info'=>'备份文件删除成功!','status'=>1]);
                //$this->success('备份文件删除成功!');
            }
        } else {
            //$this->error('参数错误!');
            $this->ajaxReturn(['info'=>'参数错误!','status'=>0]);
        }
    }

    /**
     * 备份数据库
     * @param  String  $tables 表名
     * @param  Integer $id     表ID
     * @param  Integer $start  起始行数
     */
    public function export($tables = null, $id = null, $start = null){
        if(IS_POST && !empty($tables) && is_array($tables)){ //初始化
            //读取备份配置
            $config = array(
                'path'     => realpath(C('DB_PATH')) . DIRECTORY_SEPARATOR,  //路径
                'part'     => C('DB_PART'),  //分卷大小 20M
                'compress' => C('DB_COMPRESS'),  //0:不压缩 1:启用压缩
                'level'    => C('DB_LEVEL'),  //压缩级别, 1:普通 4:一般  9:最高
            );
            //检查是否有正在执行的任务
            $lock = "{$config['path']}backup.lock";
            if(is_file($lock)){
                $this->error('检测到有一个备份任务正在执行，请稍后再试!');
            } else {
                //创建锁文件
                file_put_contents($lock, NOW_TIME);
            }

            //检查备份目录是否可写 创建备份目录
            is_writeable($config['path']) || mkdir('./'.C("DB_PATH_NAME").'',0777,true);
            session('backup_config', $config);

            //生成备份文件信息
            $file = array(
                'name' => date('Ymd-His', NOW_TIME),
                'part' => 1,
            );
            session('backup_file', $file);

            //缓存要备份的表
            session('backup_tables', $tables);
//             var_dump($file,$config);
            //创建备份文件
            $Database = new Database($file, $config);
            if(false !== $Database->create()){
//                var_dump(11111111111);
                $tab = array('id' => 0, 'start' => 0);
                $this->success('初始化成功!', '', array('tables' => $tables, 'tab' => $tab));
            } else {
//                $this->error('初始化失败，备份文件创建失败!');
            }
        } elseif (IS_GET && is_numeric($id) && is_numeric($start)) { //备份数据
            $tables = session('backup_tables');
            //备份指定表
            $Database = new Database(session('backup_file'), session('backup_config'));
            $start  = $Database->backup($tables[$id], $start);
            if(false === $start){ //出错
                $this->error('备份出错!');
            } elseif (0 === $start) { //下一表
                if(isset($tables[++$id])){
                    $tab = array('id' => $id, 'start' => 0);
                    $this->success('备份完成!', '', array('tab' => $tab));
                } else { //备份完成，清空缓存
                    //记录操作到日志
                    $action_name = "完成了一次对".$tables."表的备份";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    unlink(session('backup_config.path') . 'backup.lock');
                    session('backup_tables', null);
                    session('backup_file', null);
                    session('backup_config', null);
                    $this->success('备份完成!');
                }
            } else {
                $tab  = array('id' => $id, 'start' => $start[0]);
                $rate = floor(100 * ($start[0] / $start[1]));
                $this->success("正在备份...({$rate}%)", '', array('tab' => $tab));
            }

        } else { //出错
            $this->error('参数错误!');
        }
    }

    /**
     * 还原数据库
     */
    public function import($time = 0, $part = null, $start = null){
        if(is_numeric($time) && is_null($part) && is_null($start)){ //初始化
            //获取备份文件信息
            $name  = date('Ymd-His', $time) . '-*.sql*';
            $path  = realpath(C('DB_PATH')). DIRECTORY_SEPARATOR . $name;
            $files = glob($path);
            $list  = array();
            foreach($files as $name){
                $basename = basename($name);
                $match    = sscanf($basename, '%4s%2s%2s-%2s%2s%2s-%d');
                $gz       = preg_match('/^\d{8,8}-\d{6,6}-\d+\.sql.gz$/', $basename);
                $list[$match[6]] = array($match[6], $name, $gz);
            }
            ksort($list);

            //检测文件正确性
            $last = end($list);
            if(count($list) === $last[0]){
                session('backup_list', $list); //缓存备份列表
                $this->success('初始化完成!', '', array('part' => 1, 'start' => 0));
            } else {
                $this->error('备份文件可能已经损坏，请检查!');
            }
        } elseif(is_numeric($part) && is_numeric($start)) {
            $list  = session('backup_list');
            $db = new Database($list[$part], array(
                'path'     => realpath(C('DB_PATH')) . DIRECTORY_SEPARATOR,
                'compress' => $list[$part][2]));

            $start = $db->import($start);

            if(false === $start){
                $this->error('还原数据出错!');
            } elseif(0 === $start) { //下一卷
                if(isset($list[++$part])){
                    $data = array('part' => $part, 'start' => 0);
                    $this->success("正在还原...#{$part}", '', $data);
                } else {
                    //记录操作到日志
                    $action_name = "完成了一次数据库还原";
                    $time = date("Y-m-d H:i:s", time());
                    parent::rizhi($action_name, $time);

                    session('backup_list', null);
                    $this->success('还原完成!');
                }
            } else {
                $data = array('part' => $part, 'start' => $start[0]);
                if($start[1]){
                    $rate = floor(100 * ($start[0] / $start[1]));
                    $this->success("正在还原...#{$part} ({$rate}%)", '', $data);
                } else {
                    $data['gz'] = 1;
                    $this->success("正在还原...#{$part}", '', $data);
                }
            }

        } else {
            $this->error('参数错误!');
        }
    }

    /*
     *
     * 服务器管理
     *
     * */
    public function servers(){

        // 导入分页类
        import('Org.Util.Page');

        //查询满足满足条件的总记录条数
        $count = M("servers")->where("pid!='0'")->count();

        //实例化分页类 传入总记录数和每页显示的记录数
        $Page = new \Think\Page($count, 10);

        //分页显示输出
        $show = $Page->show();

        $data = M("servers")->where("pid!='0'")->limit($Page->firstRow . ',' . $Page->listRows)->select();
        $this->assign("arr",$data);
        $this->assign("page",$show);

        foreach ($data as $key=>$val){
            if($data[$key]['state']==0){
                $data[$key]['state']='正常';
            } else if($data[$key]['state']==1){
                $data[$key]['state']='被删';
            }
        }

        //所有大区 添加时用
        $result2 = M("servers")->where("pid=0")->select();
        $this->assign('arr2',$result2);

        $this->display();
    }
    public function data(){
        if(I('post.name')){
            $data['name']=I('post.name');
            $data['pid']=I('post.servers');
            if($data['pid']==0){
                $data['level']=1;
                $data['databass']='';
            } else{
                $data['level']=2;
                // mt_rand  获取随机数 mt_rand(min, max);
                $str="abcdefghijkmnpqrstuvwxyz";//设置被随机采集的字符串
                $codeLen='6';//设置生成的随机数个数
                function  str_rand($str,$codeLen){
                    $rand="";
                    for($i=0; $i<$codeLen-1; $i++){
                        $rand .= $str[mt_rand(0, strlen($str)-1)];  //如：随机数为30  则：$str[30]
                    }
                    return $rand;
                }
                $code=str_rand($str,$codeLen);
                $data['database']='cbg_'.$code.'_'.time();
            }
            $data['etime'] =  time();

            $result = M("servers")->add($data);
            if($result){
                if($data['database']!=''){
                    $ip=C("DB_HOST");
                    $dbname=C("DB_USER");
                    $dbpwd=C("DB_PWD");
//                    var_dump($data['database'],$ip,$dbname,$dbpwd);die;
                    database($data['database'],$ip,$dbname,$dbpwd);
                }
                $this->success('添加成功',U('database/servers'));
            } else{
                $this->error('添加失败');
            }
        }
    }
    public function xiu(){
        $id=I('get.id');

        $result = M("servers")->where('id='.$id)->select();
        $this->assign('result',$result);

        //所有大区  修改时用
        $result2 = M("servers")->where("pid=0")->select();
        $this->assign('result2',$result2);

        $this->display();
    }
    public function xiuok(){
        $id=I('post.id');
        $data=array(
            'pid'=>I('post.pid'),
            'name'=>I('post.name'),
            'state'=>I('post.state'),
            'etime'=>(string)strtotime(I("post.etime")),
        );

        $result = M("servers")->where('id='.$id)->save($data);
        if($result){
            $this->success('修改成功',U('database/servers'));
        } else {
            $this->error('修改失败');
        }
    }
    public function del2(){
        $id=I('get.id');
        $data['state']=1;
        $result = M("servers")->where('id='.$id)->save($data);
        if($result){
            $this->success('删除成功',U('database/servers'));
        } else {
            $this->error('删除失败');
        }
    }
}